package week_06;

import java.util.Arrays;

/**
 * 714. 买卖股票的最佳时机含手续费
 * 给定一个整数数组 prices，其中第 i 个元素代表了第 i 天的股票价格 ；整数 fee 代表了交易股票的手续费用。
 * <p>
 * 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。
 * <p>
 * 返回获得利润的最大值。
 * <p>
 * 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：prices = [1, 3, 2, 8, 4, 9], fee = 2
 * 输出：8
 * 解释：能够达到的最大利润:
 * 在此处买入 prices[0] = 1
 * 在此处卖出 prices[3] = 8
 * 在此处买入 prices[4] = 4
 * 在此处卖出 prices[5] = 9
 * 总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8
 * 示例 2：
 * <p>
 * 输入：prices = [1,3,7,5,10,3], fee = 3
 * 输出：6
 */
public class MaxProfit_Free_714 {
    public int maxProfit(int[] prices, int fee) {
        int n = prices.length;

        int[][] dp = new int[n + 1][2];
        int[] pricesInput = new int[n + 1];
        pricesInput[0] = 0;
        for (int i = 0; i < n; i++) {
            pricesInput[i + 1] = prices[i];
        }

        dp[0][0] = 0;
        dp[0][1] = -1000000000;
        for (int i = 1; i <= n; i++) {
            for (int j = 0; j < 2; j++) {
                // 三种状态转移:
                // case1: 不卖不买 rest 休息
                dp[i][j] = dp[i - 1][j];
                // case2: 如果持仓则可以卖出
                if (j == 0) dp[i][0] = Math.max(dp[i][0], dp[i - 1][1] + pricesInput[i] - fee);
                // case3: 如果空仓可以买入
                if (j == 1) dp[i][1] = Math.max(dp[i][j], dp[i - 1][0] - pricesInput[i]);

            }
        }
        int ans = Integer.MIN_VALUE;
        for (int j = 0; j <= 1; j++) {
            ans = Math.max(dp[n][j], ans);
        }
        return ans;
    }
}
